Breaking Entropy Bounds: Accelerating RL Training via MTP with Rejection Sampling
打破熵的限制：多token预测与拒绝采样加速强化学习训练
🧭 LLM强化学习训练加速 · 多token预测
Qwen Team, Alibaba Inc.
揭示RL阶段熵升高与MTP接受率的负线性关系，通过拒绝采样与TV损失将接受率提至95%，实现1.8倍训练加速
📌 工作参考价值（4 点）
- 拒绝采样替代贪婪采样：在电商对话式推荐或Agent策略生成中，当使用MTP（或多token生成）加速推理时，若模型因在线学习（如RL更新）导致输出分布偏移，应使用概率拒绝采样而非贪婪接受，以保持高接受率和加速比。
- 端到端优化接受率指标：推荐或生成模型中若采用多步草稿验证机制（如Sequential Recommendation中的beam search加速），可设计类似TV loss的目标函数，直接最大化多步联合接受率，替代中间步骤的KL或交叉熵损失，可提升整体推理吞吐。
- 预训练加速模块避免在线开销：在RL训练的rollout阶段，MTP模块可在RL开始前用端到端损失充分训练并冻结，避免在线更新引入的计算代价和分布波动，这对需要稳定服务的推荐Agent尤其有用。
- 熵监控指导加速策略：实际部署中，可监测生成熵值动态调整MTP步长或是否启用加速，当熵上升时自动切换到拒绝采样或回退到逐token生成，平衡效率与质量。